import os

class Config:
    
    BASE_SEARCH_URL = "https://search.bilibili.com/all?keyword=%E5%A4%A7%E8%AF%AD%E8%A8%80%E6%A8%A1%E5%9E%8B&order=totalrank"
    TARGET_VIDEOS = 300
    
    # Chrome选项
    CHROME_OPTIONS = {
        '--user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
        '--disable-blink-features': 'AutomationControlled',
        '--no-sandbox': None,
        '--disable-dev-shm-usage': None
    }
    
    # 过滤噪声弹幕
    NOISE_PATTERNS = [
        r'^666+$', r'^[0-9]{1,3}赞$', r'^点赞$', r'^哈哈哈+$',
        r'^233+$', r'^awsl$', r'^前排$', r'^打卡$', r'^签到$',
        r'^来了$', r'^第一$', r'^沙发$', r'^[0-9]+$', r'^\.+$',
        r'^开始游戏$', r'^test$', r'^三连$',r'^已三连$', r'^支持$',r'^视频$',r'^已经$',r'^弹幕$',
        r'^还是$',r'^东西$',r'^火钳$',r'^刘明$',r'^就是$',r'^硬币$',r'^可爱$',r'^投币$',r'^已经$',r'^来了$'r'^出来$'
    ]
    
    # 词云停用词
    STOP_WORDS = {
        '的', '了', '是', '在', '我', '你', '他', '她', '它', '这', '那', 
        '就', '都', '不', '也', '还', '有', '和', '与', '或', '一个', '一下',
        '这个', '那个', '什么', '怎么', '为什么', '因为', '所以', '但是', '然后',
        '啊', '哦', '嗯', '哈', '哈哈', '哈哈哈', 'hhhh', '？？', '！！！', '......',
        '开始游戏','三连','已三连','支持','视频','已经','弹幕','还是','东西','火钳','刘明','就是','硬币','可爱','投币','已经','来了','出来'
        '这么','直接','应该','我们','自己','可以','不是','这样','这么','的话','一定'
    }
    
    OUTPUT_DIR = 'output'
    
    @classmethod
    def create_output_dir(cls):
        os.makedirs(cls.OUTPUT_DIR, exist_ok=True)